import json

project_path = r"H:\wangjianlian\project\Python\dcgan"
parameters_path = project_path + r"\config\parameters.json"

class Params(object):
    '''
    主要参数类

    BATCH_SIZE : 批次数量
    '''
    def __init__(self):
        self.BATCH_SIZE = 1
        self.MODE = "train"

    def load_config_file(self, filename = parameters_path):
        # 读取数据
        with open(filename, 'r') as f:
            data = json.load(f)
        self.BATCH_SIZE = data[2]["BATCH_SIZE"]
        self.MODE = data[1]["MODE"]
